import os

sum1 = 0
# 自定义一个计算文件大小的函数
def dirsize(dir):
    '''
    计算文件大小的函数
    '''
    # 文件大小
    global sum1
    # 获取目录的所有文件信息
    dirlist = os.listdir(dir)
    # 遍历所有文件，计算大小
    for f in dirlist:
        # 获取路径
        a = os.path.join(dir,f)
        # 判断是否是文件
        if os.path.isfile(a):
            sum1 = sum1 + os.path.getsize(a)
        # 判断是否是目录
        if os.path.isdir(a):
            dirsize(a)

# 测试
dirsize(input("输入路径："))
print("文件大小为：",sum1)